tensor([[0.3100, 0.3100, 0.3100, 0.3100, 0.3100],
        [0.6900, 0.6900, 0.6900, 0.6900, 0.6900]], grad_fn=<RoundBackward1>)